from selenium.webdriver.common.by import By

from po_study.base.Base import Base


class LoginPage(Base):
    def __init__(self, driver):
        Base.__init__(self, driver)

    def input_username(self):
        self.input((By.NAME, 'username'), 'caichang')

    def input_password(self):
        self.input((By.NAME, 'password'), 'caichang1')

    def click_login_button(self):
        self.click((By.ID, 'tijiao'))
